package com.ewili.framework.utils;

import org.apache.commons.lang3.math.NumberUtils;

import java.math.BigDecimal;
import java.util.ArrayList;
import java.util.Collections;
import java.util.HashSet;

/**
 * 数学运算工具类
 * 注：所有double型的做运算必须调用此工具类
 *
 * @author leo
 */
public final class MathUtils extends NumberUtils {

    private MathUtils() {
    }

    /**
     * 加法v1+v2
     *
     * @param v1
     * @param v2
     * @return
     */
    public static double add(double v1, double v2) {
        BigDecimal b1 = toBigDecimal(v1);
        BigDecimal b2 = toBigDecimal(v2);
        return b1.add(b2).doubleValue();
    }


    /**
     * 减法v1-v2
     *
     * @param v1
     * @param v2
     * @return
     */
    public static double subtract(double v1, double v2) {
        BigDecimal b1 = toBigDecimal(v1);
        BigDecimal b2 = toBigDecimal(v2);
        return b1.subtract(b2).doubleValue();
    }

    /**
     * 乘法v1*v2
     *
     * @param v1
     * @param v2
     * @return
     */
    public static double multiply(double v1, double v2) {
        BigDecimal b1 = toBigDecimal(v1);
        BigDecimal b2 = toBigDecimal(v2);
        return b1.multiply(b2).doubleValue();
    }

    /**
     * 乘法v1*v2
     *
     * @param v1
     * @param v2
     * @param scale 保留小数点后几位小数
     * @return
     */
    public static double multiply(double v1, double v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b1 = toBigDecimal(v1);
        BigDecimal b2 = toBigDecimal(v2);
        return b1.multiply(b2).setScale(scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 除法 v1/v2
     * 如果结果出现无限循环小数会产生异常
     *
     * @param v1
     * @param v2 不能为0
     * @return
     */
    public static double divide(double v1, double v2) {
        BigDecimal b1 = toBigDecimal(v1);
        BigDecimal b2 = toBigDecimal(v2);
        return b1.divide(b2).doubleValue();
    }

    /**
     * 除法 v1/v2
     *
     * @param v1
     * @param v2
     * @param scale 保留小数点后几位小数
     * @return
     */
    public static double divide(double v1, double v2, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b1 = toBigDecimal(v1);
        BigDecimal b2 = toBigDecimal(v2);
        return b1.divide(b2, scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    /**
     * 计算百分比
     *
     * @param v1    v2
     * @param scale
     * @return
     */
    public static double percent(double v1, double v2, int scale) {
        double subtract = subtract(v1, v2);
        double divide = divide(subtract, v2, 6);
        //return divide;
        return MathUtils.multiply(divide, 100);
    }

    /**
     * 四舍五入
     *
     * @param v
     * @param scale
     * @return
     */
    public static double round(double v, int scale) {
        if (scale < 0) {
            throw new IllegalArgumentException(
                    "The scale must be a positive integer or zero");
        }
        BigDecimal b = toBigDecimal(v);
        return b.setScale(scale, BigDecimal.ROUND_HALF_UP).doubleValue();
    }

    private static BigDecimal toBigDecimal(double d) {
        return new BigDecimal(Double.toString(d));
    }

    /**
     * 穷举
     *
     * @param str
     * @return
     */
    public static ArrayList<String> permutation(String str) {
        ArrayList<String> res = new ArrayList<String>();
        if (str == null || str.length() <= 0) {
            return res;
        }
        HashSet<String> set = new HashSet<String>();
        dfs(set, str.toCharArray(), 0);
        res.addAll(set);
        Collections.sort(res);
        return res;
    }

    private static void dfs(HashSet<String> set, char[] str, int k) {
        if (k == str.length) {
            set.add(new String(str));
            return;
        }
        for (int i = 0; i < str.length; i++) {
            swap(i, k, str);
            dfs(set, str, k + 1);
            swap(i, k, str);
        }
    }

    private static void swap(int i, int j, char[] str) {
        if (i != j) {
            char temp = str[i];
            str[i] = str[j];
            str[j] = temp;
        }
    }


}
